# -*- coding: utf-8 -*-

from mc_pt2pt import PtCkpt

if __name__ == '__main__':
    src_dir = '/data/test_src'
    dst_dir = '/data/test_dst'
    iteration = 1
    
    layers = 5
    experts = 128
    
    src_tp, src_pp, src_ep, src_vpp = 4, 1, 1, 1
    dst_tp, dst_pp, dst_ep, dst_vpp = 2, 2, 2, 1
    src_num_layer_list = None
    dst_num_layer_list = None
    
    self = PtCkpt(src_dir, iteration, layers, experts,
                  src_tp, src_pp, src_ep, src_vpp, src_num_layer_list)
    # dst_ckpt = PtCkpt(dst_dir, iteration, layers, experts,
    #               dst_tp, dst_pp, dst_ep, dst_vpp, dst_num_layer_list)
    
    # dst_data = dst_ckpt.get_dst_data(0, 0, 0, src_ckpt)
    
    # dst_ckpt.run(src_ckpt)